from gensim.models import Word2Vec
import nltk
from nltk.corpus import gutenberg
nltk.download('gutenberg')
nltk.download('punkt')
macbeth = gutenberg.sents('shakespeare-macbeth.txt')
model = Word2Vec(sentences=macbeth, size=100, window= 4, min_count=10, workers=4, iter=10)